package com.t282.crm.web.controller;

import com.t282.crm.config.shiro.MyShiroRealm;
import com.t282.crm.entity.Right;
import com.t282.crm.entity.Role;
import com.t282.crm.service.RoleService;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.mgt.RealmSecurityManager;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;

import javax.annotation.Resource;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Controller
@RequestMapping("/role")
public class RoleController {
    @Resource
    private RoleService roleService;

    @GetMapping(value = "/json")
    @ResponseBody
    public List<Role> findAllRoles(){
        List<Role> roles = roleService.findAllRoles();
        return  roles;
    }

    @RequestMapping(value = "/list")
    public String list(@RequestParam(required = false,defaultValue = "") String roleName, @RequestParam(required = false,defaultValue = "1") Integer pageIndex, Model model){
        Sort sort = new Sort(Sort.Direction.ASC,"roleId");
        Pageable pageable = PageRequest.of(pageIndex-1,3,sort);
        Page<Role> rolePage = roleService.findRole(roleName,pageable);
        model.addAttribute("rolePage",rolePage);
        model.addAttribute("roleName",roleName);
        model.addAttribute("totalElements",rolePage.getTotalElements()); //总记录数
        model.addAttribute("totalPages",rolePage.getTotalPages()); //总页数
        model.addAttribute("number",rolePage.getNumber()+1);//当前页数
        return "role/rolelist";
    }

    @RequestMapping(value = "/add")
    public String add(Model model){
        //查询所有权限，用于指派权限
        List<Right> rights = roleService.findAllRights();
        model.addAttribute("rights",rights);
        return "role/add";
    }

    @RequestMapping(value = "/edit")
    public String edit(Long roleId,Model model){
        Role role = roleService.getRole(roleId);
        model.addAttribute("role",role);

        //查找角色拥有的权限
        List<Right> roleRights = roleService.findRightsByRole(role);
        model.addAttribute("roleRights",roleRights);

        //查找所有权限，用于指派权限
        List<Right> rights = roleService.findAllRights();
        model.addAttribute("rights",rights);
        return "role/edit";
    }

    @RequestMapping(value = "/save")
    public String save(Role role,String[] rightCodes){
        if(rightCodes != null){
            for (String rightCode : rightCodes){
                Right right = new Right();
                right.setRightCode(rightCode);
                role.getRights().add(right);    //将关联的权限添加到角色中，用于级联新增关联表的数据
            }
        }
        roleService.saveRole(role); //新增，会级联新增关联表的数据

        //清空权限缓存
        RealmSecurityManager rsm = (RealmSecurityManager) SecurityUtils.getSecurityManager();
        MyShiroRealm realm = (MyShiroRealm) rsm.getRealms().iterator().next();
        realm.clearAllCachedAuthorizationInfo();
        
        return "redirect:/role/list";
    }

    @RequestMapping(value = "/del")
    @ResponseBody
    public Map del(Long roleId){
        Map map = new HashMap();
        try {
            roleService.deleteRoleById(roleId);
            map.put("delResult",true);
        }catch (Exception e){
            e.printStackTrace();
            map.put("delResult",false);
        }
        return map;
    }
}
